【CUDA+Anaconda+Pytorch的安装】 您所在的位置:网站首页 nvidia 清洁安装 【CUDA+Anaconda+Pytorch的安装】

【CUDA+Anaconda+Pytorch的安装】

2023-03-14 22:51| 来源: 网络整理| 查看: 265

CUDA+Anaconda+Pytorch的安装 CUDA安装一、检查设备是否支持CUDA二、没有安装VS三、提示Nsight compute安装失败 Anaconda安装Pytorch安装换源

CUDA安装 一、检查设备是否支持CUDA

首先,在安装CUDA之前,请先去官网确认自己电脑的显卡是否支持安装CUDA

二、没有安装VS

若电脑没有装vs就不要勾选相应组件(visual studio intergratin),装了的可以勾选

请添加图片描述

三、提示Nsight compute安装失败

选择自定义安装并且取消勾选Nsight compute 在这里插入图片描述 在这里插入图片描述 安装结束后可以通过在命令行输入nvcc -V检测是否安装成功 在这里插入图片描述 显示该界面证明安装成功

Anaconda安装

直接搜索anaconda官网下载即可(注意不要安装在C盘,因为里面要下载很多虚拟环境,会占据大量空间) Anaconda安装成功后可以通过在Anaconda Prompt输入

conda list

检验是否安装成功 在Anaconda Prompt输入如下

conda create -n pytorch python=3.9.7 #创建一个新的虚拟环境命名为 pytorch conda activate pytorch #激活名为pytorch的虚拟环境

Anaconda 的其他操作

conda activate base #回到真实环境 conda remove -n pytorch --all #删除名为pytorch的虚拟环境 Pytorch安装 换源

因为外网下载pytorch会很慢,所以换成国内源,这里我使用清华源。

conda config --set show_channel_urls yes

window用户看不到condrac文件,所以用上面代码生成。之后在c盘,用户文件夹里面就能找到在这里插入图片描述 把文件夹内原始内容删掉,然后换成下面的内容,直接复制粘贴进去,记得保存!(用记事本打开即可)

channels: - defaults show_channel_urls: true default_channels: - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2 - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ custom_channels: conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud pytorch-lts: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

换源完成后 进入pytorch官网查看你要安装的版本号复制代码在虚拟环境中安装即可 例

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

在这里插入图片描述 安装结束后Anaconda Prompt中输入,如果没有报错,那么恭喜你,安装成功啦!

conda activate pytorch python import torch


【本文地址】

公司简介

联系我们

今日新闻

    推荐新闻

    专题文章
      CopyRight 2018-2019 实验室设备网 版权所有